from function import *

file_train = ".\Exp_04_train.csv"
file_test = ".\Exp_04_test.csv"

x_train, y_train = load_data(file_train)
x_test, y_test = load_data(file_test)

max_depth = [1, 2, 3]
criterion = ["entropy", "gini"]

for cri in criterion:
    for i in max_depth:
        Tree_clf(x_train, y_train, x_test, y_test, cri, i)
